import matplotlib.pyplot as plt
import numpy as np

# 三维图片，有彩色
img = np.random.rand(10,10,3)
plt.imshow(img)
plt.show()

# 保存图片到本目录中
plt.imsave('test.png',img)

plt.show()

